#pragma once
#include <stdio.h>
#include <stdlib.h>
#include <dirent.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <string.h>
#include <unistd.h>
#define PATH_MAX 4096
/// @brief 复制文件操作
/// @param src 源文件
/// @param des 目标文件
void copyfile(const char *src, const char *des)
{
    FILE *fp_src = NULL, *fp_des = NULL;
    char buf[128]; // 增大缓冲区提高效率
    size_t num;

    // 打开源文件（二进制模式确保跨平台一致性）
    if ((fp_src = fopen(src, "rb")) == NULL)
    {
        perror("Source open failed");
        exit(EXIT_FAILURE);
    }

    // 打开目标文件（二进制写入模式）
    if ((fp_des = fopen(des, "wb")) == NULL)
    {
        perror("Destination open failed");
        fclose(fp_src); // 关闭已打开的源文件
        exit(EXIT_FAILURE);
    }

    // 高效复制循环  ，fscanf(),fprintf() 可以实现复制功能
    while ((num = fread(buf, 1, sizeof(buf), fp_src)) > 0)
    {
        fwrite(buf, 1, num, fp_des);
    }

    // 关闭文件
    fclose(fp_src);
    fclose(fp_des);
}

void copy_dir(const char *src, const char *dest)
{
    struct stat st;
    if (stat(src, &st) != 0)
    {
        perror("stat failed");
        exit(EXIT_FAILURE);
    }
    if (mkdir(dest, st.st_mode) != 0)
    {
        perror("mkdir failed");
        exit(EXIT_FAILURE);
    }

    DIR *dir = opendir(src);
    if (dir == NULL)
    {
        perror("opendir failed");
        exit(EXIT_FAILURE);
    }

    struct dirent *entry;
    while ((entry = readdir(dir)) != NULL)
    {
        if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
            continue;

        char src_path[PATH_MAX]; // 使用固定大小数组
        char dest_path[PATH_MAX];
        snprintf(src_path, sizeof(src_path), "%s/%s", src, entry->d_name);
        snprintf(dest_path, sizeof(dest_path), "%s/%s", dest, entry->d_name);

        struct stat entry_st;
        if (stat(src_path, &entry_st) != 0)
        {
            perror("stat entry failed");
            continue;
        }

        if (S_ISDIR(entry_st.st_mode))
        { // 兼容性更好的目录判断
            copy_dir(src_path, dest_path);
        }
        else
        {
            copyfile(src_path, dest_path);
        }
    }
    closedir(dir);
}